from django.db import models

class TrainMedicalData(models.Model):
    date = models.DateField(null=True, blank=True)
    year = models.IntegerField()
    week_number = models.IntegerField()
    densitometer = models.IntegerField(null=True, blank=True)
    ct = models.IntegerField()
    ct_with_cu_1_zone = models.IntegerField(null=True, blank=True)
    ct_with_cu_2_or_more_zones = models.IntegerField(null=True, blank=True)
    mmg = models.IntegerField()
    mri = models.IntegerField()
    mri_with_cu_1_zone = models.IntegerField(null=True, blank=True)
    mri_with_cu_2_or_more_zones = models.IntegerField(null=True, blank=True)
    rg = models.IntegerField()
    fluorograph = models.IntegerField(null=True, blank=True)

    def __str__(self):
        return f"{self.year} - Week {self.week_number}"
    
class ValMedicalData(models.Model):
    date = models.DateField(null=True, blank=True)
    year = models.IntegerField()
    week_number = models.IntegerField()
    densitometer = models.IntegerField(null=True, blank=True)
    ct = models.IntegerField()
    ct_with_cu_1_zone = models.IntegerField(null=True, blank=True)
    ct_with_cu_2_or_more_zones = models.IntegerField(null=True, blank=True)
    mmg = models.IntegerField()
    mri = models.IntegerField()
    mri_with_cu_1_zone = models.IntegerField(null=True, blank=True)
    mri_with_cu_2_or_more_zones = models.IntegerField(null=True, blank=True)
    rg = models.IntegerField()
    fluorograph = models.IntegerField(null=True, blank=True)

    def __str__(self):
        return f"{self.year} - Week {self.week_number}"
